import util.tool as tool
from lib.llm import llm

from langchain.prompts import (
    SystemMessagePromptTemplate,
    HumanMessagePromptTemplate,
    ChatPromptTemplate,
)


# 摘要
summary = tool.read_file("data/摘要.md")

print("summary", summary)


sys_tmpl = """
你是一位资深的JSON数据转换专家，专注于将各类文档转换为符合Web前端规范的JSON格式数据。你的任务是将摘要信息转换为JSON格式，方便前端组件调用。

任务目标：
JSON字段使用纯中文
保持原始数据的完整性和准确性
输出符合网页直接使用的规范JSON

要求：

输出结果：
请严格按照JSON格式返回结果，只需给出JSON内容，不需要其他说明：
"""


sys_msg = SystemMessagePromptTemplate.from_template(sys_tmpl)
user_msg = HumanMessagePromptTemplate.from_template("{summary}")

# print("sys_tmpl", sys_tmpl)

# 组合聊天提示
prompt = ChatPromptTemplate.from_messages([sys_msg, user_msg])
formatted_prompt = prompt.format(summary=summary)

# 对话模型调用
response = llm.invoke(formatted_prompt)

print(response.content)
tool.save_file("data/解析摘要2.json", response.content)
